import numpy as np
import pandas as pd


def wirte():
    df = pd.DataFrame({
        "学院": ["电子学院", "机电学院", "大数据学院", "软件学院", "网络学院"],
        "2017年": [1320, 1351, 2251, 3178, 2151],
        "2018年": [1323, 1402, 2202, 3180, 2190],
        "2019年": [1466, 1322, 2222, 3162, 2122],
        "2020年": [1486, 1388, 2248, 3218, 3108]
    })

    # index=False表示不写入索引
    df.to_csv("学院.csv", index=False)


def read():
    # usecols获取指定列数据
    data = pd.read_csv(r'学院.csv', delimiter=',', usecols=["2017年", '2018年', '2019年', '2020年'])
    print(data)
    print(data.shape)
    # 得到ndarray
    print(data.values)
    print(data.values.shape)
    # 转换矩阵
    print(data.values.reshape(-1))
    # 排序
    print(np.sort(data.values.reshape(-1)))

    x, y = pd.read_csv(r'学院.csv', delimiter=',', usecols=["2017年", '2018年'])
    print(f"x:{x} ------> y:{y}")


if __name__ == '__main__':
    # wirte()
    read()
